#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <ctemplate/template.h>

#include "oj_model.hpp"

namespace ns_view
{

    // struct Question
    // {
    //     std::string number; // 题目编号
    //     std::string title;  // 题目标题
    //     std::string star;   // 难度：简单，中等，困难
    //     int cpu_limit;      // 时间限制：cpu_limit: s
    //     int mem_limit;      // 空间限制：mem_limit: KB

    //     std::string desc;   // 题目描述
    //     std::string header; // 在线编辑器
    //     std::string tail;   // 测试用例
    // };

    const std::string template_path = "./template_html/";

    using namespace ns_model;

    class View
    {

    public:
        // 编号，标题，难度
        // 使用表格显示
        static bool AllExpandHtml(const std::vector<Question> &questions, std::string *html)
        {
            // 1.形成路径
            std::string src_html = template_path + "all_questions.html";

            // 2. 形成数字字典
            ctemplate::TemplateDictionary root("all_questions");
            for (const auto &q : questions)
            {
                ctemplate::TemplateDictionary *sub = root.AddSectionDictionary("question_list");
                sub->SetValue("number", q.number);
                sub->SetValue("title", q.title);
                sub->SetValue("star", q.star);
            }
            // 3. 获取被渲染的网页
            ctemplate::Template *tpl = ctemplate::Template::GetTemplate(src_html, ctemplate::DO_NOT_STRIP); // 不做修改

            // 4. 开始完成渲染
            tpl->Expand(html, &root);
            return true;
        }

        static bool OneExpandHtml(const Question &q, std::string *html)
        {
            std::string src_html = template_path + "one_question.html";

            // 数字字典
            ctemplate::TemplateDictionary root("one_question");

            root.SetValue("number", q.number);
            root.SetValue("title", q.title);
            root.SetValue("star", q.star);
            root.SetValue("desc", q.desc);
            root.SetValue("pre_code", q.header);

            // 获取要渲染的网页
            ctemplate::Template *tpl = ctemplate::Template::GetTemplate(src_html, ctemplate::DO_NOT_STRIP); // 不做修改

            // 渲染
            tpl->Expand(html, &root);

            return true;
        }
    };
}